import torch
if torch.cuda.is_available():
    print('CUDA is available!')
else:
    print('CUDA is not available.')
print("CUDA runtime version:", torch.version.cuda)
print("cuDNN version:", torch.backends.cudnn.version())